package com.xcc.server.hadler;

import com.alibaba.fastjson.JSON;
import com.xcc.server.Server;
import com.xcc.server.domains.enums.MessageType;
import com.xcc.server.domains.vo.Message;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.util.Attribute;
import io.netty.util.AttributeKey;
import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang3.StringUtils;

import java.util.Map;

/**
 * @Author: xcc
 * @Date: 2021/2/5 14:28
 * @Description: 基础处理器
 */
public abstract class BaseHandler extends SimpleChannelInboundHandler<Message<String>> {

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, Message<String> msg) throws Exception {
        // 区分登录和数据传输
        // 如果是登录, 加入到长连接列表中
        System.out.println("收到消息: " + JSON.toJSONString(msg));
        Channel channel = ctx.channel();
        MessageType messageType = msg.getMessageType();
        if (null == messageType) {
            // 没有获取到消息类型
            System.out.println("无消息类型");
            return;
        }

        // 登录
        if (MessageType.LOGIN.equals(messageType)) {
            Map<String, Object> header = msg.getHeader();
            if (MapUtils.isEmpty(header)) {
                System.err.println("未获取到请求头");
                return;
            }
            Object token = header.get("token");
            if (null == token) {
                System.err.println("未获取到token");
                return;
            }
            AttributeKey<String> auth = AttributeKey.valueOf("token");
            Attribute<String> attr = channel.attr(auth);
            if (StringUtils.isBlank(attr.get())) { // 首次登录,
                // TODO: 添加身份校验
                channel.attr(auth).set(token.toString());
                Server.CHANNEL_MAP.putIfAbsent(token.toString(), ctx);
                System.out.println("当前登录用户: " + token);
            }
            return;
        }

        if (!MessageType.BODY.equals(messageType)) {
            // 其它, 不处理交给心跳链接处理
            System.out.println("其它消息");
            return;
        }
        // 数据交互
        System.out.println("收到数据内容: " + msg.getBody());
    }
}
